package org.dromara.common.mybatis.helper

import cn.hutool.core.convert.Convert
import com.baomidou.dynamic.datasource.DynamicRoutingDataSource
import org.dromara.common.core.exception.ServiceException
import org.dromara.common.core.utils.SpringUtils.getBean
import org.dromara.common.mybatis.enums.DataBaseType
import org.dromara.common.mybatis.enums.DataBaseType.Companion.find
import java.sql.SQLException

/**
 * 数据库助手
 *
 * @author Lion Li
 * @updater LikeYouDo
 * @date 2025/1/27 3:19
 */
object DataBaseHelper {
    private val DS: DynamicRoutingDataSource = getBean(DynamicRoutingDataSource::class.java)

    /**
     * 获取当前数据源对应的数据库类型
     * <p>
     * 通过 DynamicRoutingDataSource 获取当前线程绑定的数据源，
     * 然后从数据源获取数据库连接，利用连接的元数据获取数据库产品名称，
     * 最后调用 DataBaseType.find 方法将数据库名称转换为对应的枚举类型
     *
     * @return 当前数据库对应的 DataBaseType 枚举，找不到时默认返回 MY_SQL
     * @throws ServiceException 当获取数据库连接或元数据出现异常时抛出业务异常
     */
    @JvmStatic
    fun getDataBaseType(): DataBaseType? {
        val dataSource = DS.determineDataSource()
        try {
            dataSource.connection.use { conn ->
                val metaData = conn.metaData
                val databaseProductName = metaData.databaseProductName
                return find(databaseProductName)
            }
        } catch (e: SQLException) {
            throw ServiceException(e.message)
        }
    }

    /**
     * 根据当前数据库类型，生成兼容的 FIND_IN_SET 语句片段
     * <p>
     * 用于判断指定值是否存在于逗号分隔的字符串列中，SQL写法根据不同数据库方言自动切换：
     * - Oracle 使用 instr 函数
     * - PostgreSQL 使用 strpos 函数
     * - SQL Server 使用 charindex 函数
     * - 其他默认使用 MySQL 的 find_in_set 函数
     *
     * @param var1 要查找的值（支持任意类型，内部会转换成字符串）
     * @param var2 存储逗号分隔值的数据库列名
     * @return 适用于当前数据库的 SQL 条件字符串，通常用于 where 或 apply 中拼接
     */
    @JvmStatic
    fun findInSet(var1: Any?, var2: String?): String {
        val value = Convert.toStr(var1)
        return when (getDataBaseType()) {
            // instr(',0,100,101,' , ',100,') <> 0
            DataBaseType.ORACLE -> "instr(','||$var2||',' , ',$value,') <> 0"
            // (select strpos(',0,100,101,' , ',100,')) <> 0
            DataBaseType.POSTGRE_SQL -> "(select strpos(','||$var2||',' , ',$value,')) <> 0"
            // charindex(',100,' , ',0,100,101,') <> 0
            DataBaseType.SQL_SERVER -> "charindex(',$value,' , ','+$var2+',') <> 0"
            // find_in_set(100 , '0,100,101')
            else -> "find_in_set('$value' , $var2) <> 0"
        }
    }

    /**
     * 获取当前加载的数据库名
     */
    @JvmStatic
    fun getDataSourceNameList(): List<String> {
        return ArrayList(DS.dataSources.keys)
    }
}
